/* USER CODE BEGIN Header */
/**
  ******************************************************************************
  * @file    usart.c
  * @brief   This file provides code for the configuration
  *          of the USART instances.
  ******************************************************************************
  * @attention
  *
  * Copyright (c) 2024 STMicroelectronics.
  * All rights reserved.
  *
  * This software is licensed under terms that can be found in the LICENSE file
  * in the root directory of this software component.
  * If no LICENSE file comes with this software, it is provided AS-IS.
  *
  ******************************************************************************
  */
#include <math.h>
#include <string.h>
#include <stdio.h>
#include "sensor.h"
/* USER CODE END Header */
/* Includes ------------------------------------------------------------------*/
#include "usart.h"

/* USER CODE BEGIN 0 */

typedef HAL_StatusTypeDef (*dev_uart_send_data)(UART_HandleTypeDef *huart, const uint8_t *pData, uint16_t Size);

int __io_putchar(int ch)
{
	 HAL_UART_Transmit(&huart2, (uint8_t *)&ch, 1, 1000);
	 return ch;
}

typedef struct
{
    struct
    {
        uint8_t id;
        UART_HandleTypeDef *handle;
    }uart;
    dev_uart_send_data send;
    usart_buf_t *buf;
    uart_rcv_callback cb;
}dev_uart_t;



usart_buf_t usart_buf[UART_ID_NUM];

static dev_uart_t  s_uart[UART_ID_NUM] = {{{1, &huart1}, HAL_UART_Transmit_DMA, &usart_buf[UART_ID_SENSOR]}, \
                                          {{2, &huart2}, HAL_UART_Transmit_DMA, &usart_buf[UART_ID_MODBUS]}, \
                                         };

/* USER CODE END 0 */

UART_HandleTypeDef huart1;
UART_HandleTypeDef huart2;
DMA_HandleTypeDef hdma_usart1_rx;
DMA_HandleTypeDef hdma_usart1_tx;
DMA_HandleTypeDef hdma_usart2_rx;
DMA_HandleTypeDef hdma_usart2_tx;

/* USART1 init function */

void MX_USART1_UART_Init(void)
{

  /* USER CODE BEGIN USART1_Init 0 */

  /* USER CODE END USART1_Init 0 */

  /* USER CODE BEGIN USART1_Init 1 */

  /* USER CODE END USART1_Init 1 */
  huart1.Instance = USART1;
  huart1.Init.BaudRate = 9600;
  huart1.Init.WordLength = UART_WORDLENGTH_8B;
  huart1.Init.StopBits = UART_STOPBITS_1;
  huart1.Init.Parity = UART_PARITY_NONE;
  huart1.Init.Mode = UART_MODE_TX_RX;
  huart1.Init.HwFlowCtl = UART_HWCONTROL_NONE;
  huart1.Init.OverSampling = UART_OVERSAMPLING_16;
  huart1.Init.OneBitSampling = UART_ONE_BIT_SAMPLE_DISABLE;
  huart1.Init.ClockPrescaler = UART_PRESCALER_DIV1;
  huart1.AdvancedInit.AdvFeatureInit = UART_ADVFEATURE_NO_INIT;
  if (HAL_UART_Init(&huart1) != HAL_OK)
  {
    Error_Handler();
  }
  if (HAL_UARTEx_SetTxFifoThreshold(&huart1, UART_TXFIFO_THRESHOLD_1_8) != HAL_OK)
  {
    Error_Handler();
  }
  if (HAL_UARTEx_SetRxFifoThreshold(&huart1, UART_RXFIFO_THRESHOLD_1_8) != HAL_OK)
  {
    Error_Handler();
  }
  if (HAL_UARTEx_DisableFifoMode(&huart1) != HAL_OK)
  {
    Error_Handler();
  }
  /* USER CODE BEGIN USART1_Init 2 */
  __HAL_UART_ENABLE_IT(&huart1, UART_IT_IDLE);   //开启IDLE中断，以帧方式接收不定长数据
  HAL_UARTEx_ReceiveToIdle_DMA(&huart1, usart_buf[UART_ID_SENSOR].rx_buf, USART_BUFF_SIZE);  //开始DMA接收
  /* USER CODE END USART1_Init 2 */

}
/* USART2 init function */

void MX_USART2_UART_Init(void)
{

  /* USER CODE BEGIN USART2_Init 0 */

  /* USER CODE END USART2_Init 0 */

  /* USER CODE BEGIN USART2_Init 1 */

  /* USER CODE END USART2_Init 1 */
  huart2.Instance = USART2;
  huart2.Init.BaudRate = 115200;
  huart2.Init.WordLength = UART_WORDLENGTH_8B;
  huart2.Init.StopBits = UART_STOPBITS_1;
  huart2.Init.Parity = UART_PARITY_NONE;
  huart2.Init.Mode = UART_MODE_TX_RX;
  huart2.Init.HwFlowCtl = UART_HWCONTROL_NONE;
  huart2.Init.OverSampling = UART_OVERSAMPLING_16;
  huart2.Init.OneBitSampling = UART_ONE_BIT_SAMPLE_DISABLE;
  huart2.Init.ClockPrescaler = UART_PRESCALER_DIV1;
  huart2.AdvancedInit.AdvFeatureInit = UART_ADVFEATURE_NO_INIT;
  if (HAL_UART_Init(&huart2) != HAL_OK)
  {
    Error_Handler();
  }
  /* USER CODE BEGIN USART2_Init 2 */
  __HAL_UART_ENABLE_IT(&huart2, UART_IT_IDLE);   //开启IDLE中断，以帧方式接收不定长数据
  HAL_UARTEx_ReceiveToIdle_DMA(&huart2, usart_buf[UART_ID_MODBUS].rx_buf, USART_BUFF_SIZE);  //开始DMA接收
  /* USER CODE END USART2_Init 2 */

}

void HAL_UART_MspInit(UART_HandleTypeDef* uartHandle)
{

  GPIO_InitTypeDef GPIO_InitStruct = {0};
  RCC_PeriphCLKInitTypeDef PeriphClkInit = {0};
  if(uartHandle->Instance==USART1)
  {
  /* USER CODE BEGIN USART1_MspInit 0 */

  /* USER CODE END USART1_MspInit 0 */

  /** Initializes the peripherals clocks
  */
    PeriphClkInit.PeriphClockSelection = RCC_PERIPHCLK_USART1;
    PeriphClkInit.Usart1ClockSelection = RCC_USART1CLKSOURCE_PCLK1;
    if (HAL_RCCEx_PeriphCLKConfig(&PeriphClkInit) != HAL_OK)
    {
      Error_Handler();
    }

    /* USART1 clock enable */
    __HAL_RCC_USART1_CLK_ENABLE();

    __HAL_RCC_GPIOA_CLK_ENABLE();
    /**USART1 GPIO Configuration
    PA9     ------> USART1_TX
    PA10     ------> USART1_RX
    */
    GPIO_InitStruct.Pin = GPIO_PIN_9|GPIO_PIN_10;
    GPIO_InitStruct.Mode = GPIO_MODE_AF_PP;
    GPIO_InitStruct.Pull = GPIO_NOPULL;
    GPIO_InitStruct.Speed = GPIO_SPEED_FREQ_LOW;
    GPIO_InitStruct.Alternate = GPIO_AF1_USART1;
    HAL_GPIO_Init(GPIOA, &GPIO_InitStruct);

    /* USART1 DMA Init */
    /* USART1_RX Init */
    hdma_usart1_rx.Instance = DMA1_Channel1;
    hdma_usart1_rx.Init.Request = DMA_REQUEST_USART1_RX;
    hdma_usart1_rx.Init.Direction = DMA_PERIPH_TO_MEMORY;
    hdma_usart1_rx.Init.PeriphInc = DMA_PINC_DISABLE;
    hdma_usart1_rx.Init.MemInc = DMA_MINC_ENABLE;
    hdma_usart1_rx.Init.PeriphDataAlignment = DMA_PDATAALIGN_BYTE;
    hdma_usart1_rx.Init.MemDataAlignment = DMA_MDATAALIGN_BYTE;
    hdma_usart1_rx.Init.Mode = DMA_NORMAL;
    hdma_usart1_rx.Init.Priority = DMA_PRIORITY_LOW;
    if (HAL_DMA_Init(&hdma_usart1_rx) != HAL_OK)
    {
      Error_Handler();
    }

    __HAL_LINKDMA(uartHandle,hdmarx,hdma_usart1_rx);

    /* USART1_TX Init */
    hdma_usart1_tx.Instance = DMA1_Channel2;
    hdma_usart1_tx.Init.Request = DMA_REQUEST_USART1_TX;
    hdma_usart1_tx.Init.Direction = DMA_MEMORY_TO_PERIPH;
    hdma_usart1_tx.Init.PeriphInc = DMA_PINC_DISABLE;
    hdma_usart1_tx.Init.MemInc = DMA_MINC_ENABLE;
    hdma_usart1_tx.Init.PeriphDataAlignment = DMA_PDATAALIGN_BYTE;
    hdma_usart1_tx.Init.MemDataAlignment = DMA_MDATAALIGN_BYTE;
    hdma_usart1_tx.Init.Mode = DMA_NORMAL;
    hdma_usart1_tx.Init.Priority = DMA_PRIORITY_LOW;
    if (HAL_DMA_Init(&hdma_usart1_tx) != HAL_OK)
    {
      Error_Handler();
    }

    __HAL_LINKDMA(uartHandle,hdmatx,hdma_usart1_tx);

    /* USART1 interrupt Init */
    HAL_NVIC_SetPriority(USART1_IRQn, 0, 0);
    HAL_NVIC_EnableIRQ(USART1_IRQn);
  /* USER CODE BEGIN USART1_MspInit 1 */

  /* USER CODE END USART1_MspInit 1 */
  }
  else if(uartHandle->Instance==USART2)
  {
  /* USER CODE BEGIN USART2_MspInit 0 */

  /* USER CODE END USART2_MspInit 0 */

    /* USART2 clock enable */
    __HAL_RCC_USART2_CLK_ENABLE();

    __HAL_RCC_GPIOA_CLK_ENABLE();
    /**USART2 GPIO Configuration
    PA2     ------> USART2_TX
    PA3     ------> USART2_RX
    */
    GPIO_InitStruct.Pin = GPIO_PIN_2|GPIO_PIN_3;
    GPIO_InitStruct.Mode = GPIO_MODE_AF_PP;
    GPIO_InitStruct.Pull = GPIO_NOPULL;
    GPIO_InitStruct.Speed = GPIO_SPEED_FREQ_LOW;
    GPIO_InitStruct.Alternate = GPIO_AF1_USART2;
    HAL_GPIO_Init(GPIOA, &GPIO_InitStruct);

    /* USART2 DMA Init */
    /* USART2_RX Init */
    hdma_usart2_rx.Instance = DMA1_Channel3;
    hdma_usart2_rx.Init.Request = DMA_REQUEST_USART2_RX;
    hdma_usart2_rx.Init.Direction = DMA_PERIPH_TO_MEMORY;
    hdma_usart2_rx.Init.PeriphInc = DMA_PINC_DISABLE;
    hdma_usart2_rx.Init.MemInc = DMA_MINC_ENABLE;
    hdma_usart2_rx.Init.PeriphDataAlignment = DMA_PDATAALIGN_BYTE;
    hdma_usart2_rx.Init.MemDataAlignment = DMA_MDATAALIGN_BYTE;
    hdma_usart2_rx.Init.Mode = DMA_NORMAL;
    hdma_usart2_rx.Init.Priority = DMA_PRIORITY_LOW;
    if (HAL_DMA_Init(&hdma_usart2_rx) != HAL_OK)
    {
      Error_Handler();
    }

    __HAL_LINKDMA(uartHandle,hdmarx,hdma_usart2_rx);

    /* USART2_TX Init */
    hdma_usart2_tx.Instance = DMA1_Channel4;
    hdma_usart2_tx.Init.Request = DMA_REQUEST_USART2_TX;
    hdma_usart2_tx.Init.Direction = DMA_MEMORY_TO_PERIPH;
    hdma_usart2_tx.Init.PeriphInc = DMA_PINC_DISABLE;
    hdma_usart2_tx.Init.MemInc = DMA_MINC_ENABLE;
    hdma_usart2_tx.Init.PeriphDataAlignment = DMA_PDATAALIGN_BYTE;
    hdma_usart2_tx.Init.MemDataAlignment = DMA_MDATAALIGN_BYTE;
    hdma_usart2_tx.Init.Mode = DMA_NORMAL;
    hdma_usart2_tx.Init.Priority = DMA_PRIORITY_LOW;
    if (HAL_DMA_Init(&hdma_usart2_tx) != HAL_OK)
    {
      Error_Handler();
    }

    __HAL_LINKDMA(uartHandle,hdmatx,hdma_usart2_tx);

    /* USART2 interrupt Init */
    HAL_NVIC_SetPriority(USART2_IRQn, 0, 0);
    HAL_NVIC_EnableIRQ(USART2_IRQn);
  /* USER CODE BEGIN USART2_MspInit 1 */

  /* USER CODE END USART2_MspInit 1 */
  }
}

void HAL_UART_MspDeInit(UART_HandleTypeDef* uartHandle)
{

  if(uartHandle->Instance==USART1)
  {
  /* USER CODE BEGIN USART1_MspDeInit 0 */

  /* USER CODE END USART1_MspDeInit 0 */
    /* Peripheral clock disable */
    __HAL_RCC_USART1_CLK_DISABLE();

    /**USART1 GPIO Configuration
    PA9     ------> USART1_TX
    PA10     ------> USART1_RX
    */
    HAL_GPIO_DeInit(GPIOA, GPIO_PIN_9|GPIO_PIN_10);

    /* USART1 DMA DeInit */
    HAL_DMA_DeInit(uartHandle->hdmarx);
    HAL_DMA_DeInit(uartHandle->hdmatx);

    /* USART1 interrupt Deinit */
    HAL_NVIC_DisableIRQ(USART1_IRQn);
  /* USER CODE BEGIN USART1_MspDeInit 1 */

  /* USER CODE END USART1_MspDeInit 1 */
  }
  else if(uartHandle->Instance==USART2)
  {
  /* USER CODE BEGIN USART2_MspDeInit 0 */

  /* USER CODE END USART2_MspDeInit 0 */
    /* Peripheral clock disable */
    __HAL_RCC_USART2_CLK_DISABLE();

    /**USART2 GPIO Configuration
    PA2     ------> USART2_TX
    PA3     ------> USART2_RX
    */
    HAL_GPIO_DeInit(GPIOA, GPIO_PIN_2|GPIO_PIN_3);

    /* USART2 DMA DeInit */
    HAL_DMA_DeInit(uartHandle->hdmarx);
    HAL_DMA_DeInit(uartHandle->hdmatx);

    /* USART2 interrupt Deinit */
    HAL_NVIC_DisableIRQ(USART2_IRQn);
  /* USER CODE BEGIN USART2_MspDeInit 1 */

  /* USER CODE END USART2_MspDeInit 1 */
  }
}

/* USER CODE BEGIN 1 */
//void HAL_UART_RxCpltCallback(UART_HandleTypeDef *huart)
//{
//    if(huart->Instance==USART1)
//    {
//        HAL_UART_DMAStop(huart);
//        usart_buf[UART_ID_SENSOR].rx_len=USART_BUFF_SIZE-hdma_usart2_rx.Instance->CNDTR;    //获取接收到的数据数量(G0或G4要把NDTR改成CNDTR)
//
//        if (s_uart[UART_ID_SENSOR].cb != NULL)
//            s_uart[UART_ID_SENSOR].cb(UART_ID_SENSOR, usart_buf[UART_ID_SENSOR].rx_buf, usart_buf[UART_ID_SENSOR].rx_len);
//
//#if USE_CMD_BUF
//        //把数据保存到sensor.c里定义的数据缓冲区
//        sensor_set_data(usart_buf[UART_ID_SENSOR].rx_buf, usart_buf[UART_ID_SENSOR].rx_len);
//#endif
//        /* 测试使用，收到的数据发送出去 */
////        memcpy(usart_buf[UART_ID_SENSOR].tx_buf,usart_buf[UART_ID_SENSOR].rx_buf,usart_buf[UART_ID_SENSOR].rx_len); //清空缓存，重新接收
////        HAL_UART_Transmit_DMA(&huart2,(uint8_t *)&usart_buf[UART_ID_SENSOR].tx_buf,usart_buf[UART_ID_SENSOR].rx_len);
//
//        memset(usart_buf[UART_ID_SENSOR].rx_buf,0x00,USART_BUFF_SIZE);  //清空缓存，重新接收
//        HAL_UARTEx_ReceiveToIdle_DMA(huart,(uint8_t *)&usart_buf[UART_ID_SENSOR].rx_buf,USART_BUFF_SIZE); //开启DMA接收
//    }
//    else if(huart->Instance==USART2)
//    {
//        HAL_UART_DMAStop(huart);
//        usart_buf[UART_ID_MODBUS].rx_len=USART_BUFF_SIZE-hdma_usart2_rx.Instance->CNDTR;	//获取接收到的数据数量(G0或G4要把NDTR改成CNDTR)
//
//        if (s_uart[UART_ID_MODBUS].cb != NULL)
//            s_uart[UART_ID_MODBUS].cb(UART_ID_MODBUS, usart_buf[UART_ID_MODBUS].rx_buf, usart_buf[UART_ID_MODBUS].rx_len);
//
//        /* 测试使用，收到的数据发送出去 */
//        memcpy(usart_buf[UART_ID_MODBUS].tx_buf,usart_buf[UART_ID_MODBUS].rx_buf,usart_buf[UART_ID_MODBUS].rx_len);	//清空缓存，重新接收
//        HAL_UART_Transmit_DMA(huart,(uint8_t *)&usart_buf[UART_ID_MODBUS].tx_buf,usart_buf[UART_ID_MODBUS].rx_len);
//
//        memset(usart_buf[UART_ID_MODBUS].rx_buf,0x00,USART_BUFF_SIZE);	//清空缓存，重新接收
//        HAL_UARTEx_ReceiveToIdle_DMA(huart,(uint8_t *)&usart_buf[UART_ID_MODBUS].rx_buf,USART_BUFF_SIZE); //开启DMA接收
//    }
//}

void HAL_UARTEx_RxEventCallback(UART_HandleTypeDef *huart, uint16_t Size)
{
    if(huart->Instance==USART1)
    {
        HAL_UART_DMAStop(huart);
        usart_buf[UART_ID_SENSOR].rx_len=Size;//USART_BUFF_SIZE-hdma_usart2_rx.Instance->CNDTR;    //获取接收到的数据数量(G0或G4要把NDTR改成CNDTR)

#if USE_CMD_BUF
        //把数据保存到sensor.c里定义的数据缓冲区
        sensor_set_data(usart_buf[UART_ID_SENSOR].rx_buf, Size);
#endif

        //if (s_uart[UART_ID_MODBUS].cb != NULL)
        //  s_uart[UART_ID_MODBUS].cb(UART_ID_MODBUS, usart_buf[UART_ID_MODBUS].rx_buf, usart_buf[UART_ID_MODBUS].rx_len);

        /* 测试使用，收到的数据发送出去 */
//        memcpy(usart_buf[UART_ID_SENSOR].tx_buf,usart_buf[UART_ID_SENSOR].rx_buf,usart_buf[UART_ID_SENSOR].rx_len); //清空缓存，重新接收
//        HAL_UART_Transmit_DMA(&huart2,(uint8_t *)usart_buf[UART_ID_SENSOR].tx_buf,Size);

        memset(usart_buf[UART_ID_SENSOR].rx_buf,0x00,USART_BUFF_SIZE);  //清空缓存，重新接收
        HAL_UARTEx_ReceiveToIdle_DMA(huart,(uint8_t *)usart_buf[UART_ID_SENSOR].rx_buf,USART_BUFF_SIZE); //开启DMA接收
    }
    else if(huart->Instance==USART2)
    {
        HAL_UART_DMAStop(huart);
        usart_buf[UART_ID_MODBUS].rx_len=Size;//USART_BUFF_SIZE-hdma_usart2_rx.Instance->CNDTR;    //获取接收到的数据数量(G0或G4要把NDTR改成CNDTR)

        //if (s_uart[UART_ID_MODBUS].cb != NULL)
        //  s_uart[UART_ID_MODBUS].cb(UART_ID_MODBUS, usart_buf[UART_ID_MODBUS].rx_buf, usart_buf[UART_ID_MODBUS].rx_len);

        /* 测试使用，收到的数据发送出去 */
        memcpy(usart_buf[UART_ID_MODBUS].tx_buf,usart_buf[UART_ID_MODBUS].rx_buf,usart_buf[UART_ID_MODBUS].rx_len); //清空缓存，重新接收
        HAL_UART_Transmit_DMA(huart,(uint8_t *)usart_buf[UART_ID_MODBUS].tx_buf,Size);

        memset(usart_buf[UART_ID_MODBUS].rx_buf,0x00,USART_BUFF_SIZE);  //清空缓存，重新接收
        HAL_UARTEx_ReceiveToIdle_DMA(huart,(uint8_t *)usart_buf[UART_ID_MODBUS].rx_buf,USART_BUFF_SIZE); //开启DMA接收
    }
}


void HAL_UART_ErrorCallback(UART_HandleTypeDef *huart)
{
    if(huart->Instance == USART1)
    {
        __HAL_UNLOCK(&huart1);
        memset(usart_buf[UART_ID_SENSOR].rx_buf,0x00,USART_BUFF_SIZE);  //清空缓存，重新接收
        HAL_UARTEx_ReceiveToIdle_DMA(huart,(uint8_t *)usart_buf[UART_ID_SENSOR].rx_buf,USART_BUFF_SIZE); //开启DMA接收
    }
    else if(huart->Instance == USART2)
    {
        __HAL_UNLOCK(&huart2);
        memset(usart_buf[UART_ID_MODBUS].rx_buf,0x00,USART_BUFF_SIZE);	//清空缓存，重新接收
        HAL_UARTEx_ReceiveToIdle_DMA(huart,(uint8_t *)usart_buf[UART_ID_MODBUS].rx_buf,USART_BUFF_SIZE); //开启DMA接收
    }
}

//----------------------修改波特率-------------------------------------------------------------------------------------
void UART1_SetBaudRate(uint32_t baudrate)
{
    // 直接修改波特率
//    huart1.Init.BaudRate = baudrate;
    // 初始化UART1（这将应用新的波特率）
//    HAL_UART_Init(&huart1);
}

//----------------------UART1发送数据-------------------------------------------------------------------------------------
void UART1_SendBytes(uint8_t* data, int size)
{
	HAL_UART_Transmit_DMA(&huart1, data, size);
}
//----------------------UART2发送数据-------------------------------------------------------------------------------------
void UART2_SendBytes(uint8_t* data, int size)
{
	HAL_UART_Transmit_DMA(&huart2, data, size);
}

/* USER CODE END 1 */
